Simplify your start #16
Conversation
Rinatum
commented
Jul 27, 2021
Rinatum
commented
- update requirements.txt
- add dockerfile
- add imageio.imsave
- add tqdm for main loops
requirements.txt updated dockerfile added
| torch==1.4.0 | ||
| torchvision==0.5.0 |
There was a problem hiding this comment.
Hi there ✋🏼,
I have run the train_interpreter.py file from this repo using torch and torchvision latest versions.
To do so, I have changed these two lines by:
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.9.0+cu111
torchvision==0.10.0+cu111
There was a problem hiding this comment.
Hello, Thank you
I didn't test this torch/torchvision versions because
torch==1.4.0
torchvision==0.5.0
were added by original author of this repo
Also you need to update Dockerfile to use cuda/cudann suitable for versions 1.9.0 and 0.10.0
Do you think the changes you proposed are needed?
There was a problem hiding this comment.
I am not really sure if these changes affect another functionality of this repo. So, I wouldn't include them directly.
However, from my point of view, I would try to update these versions to the latest PyTorch versions to avoid future problems (just like the "imsave" function was deprecated). Moreover, the use of recent versions would mean greater support from the PyTorch community.
